python matrix interview questions